deepseek-chat-ext/src/extension.ts at main · Eyelor/deepseek-chat-ext · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
// The module 'vscode' contains the VS Code extensibility API
// Import the module and reference it with the alias vscode in your code below
import * as vscode from 'vscode';
import ollama from 'ollama';
import { exec } from 'child_process';

// This method is called when your extension is activated
// Your extension is activated the very first time the command is executed
export function activate(context: vscode.ExtensionContext) {

	// Use the console to output diagnostic information (console.log) and errors (console.error)
	// This line of code will only be executed once when your extension is activated
	console.log('Congratulations, your extension "deepseek-chat-ext" is now active!');

	// The command has been defined in the package.json file
	// Now provide the implementation of the command with registerCommand
	// The commandId parameter must match the command field in package.json
	const disposable = vscode.commands.registerCommand('deepseek-chat-ext.start', () => {
		// The code you place here will be executed every time your command is executed
		// Display a message box to the user
		const panel = vscode.window.createWebviewPanel(
			'deepseekChat',
			'DeepSeek Chat',
			vscode.ViewColumn.One,
			{enableScripts: true}
		);

		panel.webview.html = getWebviewContent();

		panel.webview.onDidReceiveMessage( async (message: any) => {
			if (message.command === 'chat') {
				const userPrompt = message.text;
				let responseText = '';

				try {
					const streamResponse = await ollama.chat({
						model: 'deepseek-coder-v2:16b',
						messages: [{ role: 'user', content: userPrompt }],
						stream: true
					});

					for await (const part of streamResponse) {
						responseText += part.message.content;
						panel.webview.postMessage({ command: 'chatResponse', text: responseText });
					}
				} catch (err) {
					console.error(err);
					panel.webview.postMessage({ command: 'chatResponse', text: `Error: ${String(err)}, wait a few seconds and try again :)` });
					exec('ollama serve', (error, stdout, stderr) => {
                        if (error) {
                            console.error(`Error executing ollama serve: ${error.message}`);
                            return;
                        }
                        if (stderr) {
                            console.error(`stderr: ${stderr}`);
                            return;
                        }
                        console.log(`stdout: ${stdout}`);
                    });
				}
			}
		});
	});

	context.subscriptions.push(disposable);
}

function getWebviewContent(): string {
	return /*html*/`
	<!DOCTYPE html>
	<html lang="en">
	<head>
		<meta charset="UTF-8">
		<title>DeepSeek Chat</title>
		<script src="https://cdn.jsdelivr.net/npm/marked/marked.min.js"></script>
        <script src="https://cdnjs.cloudflare.com/ajax/libs/highlight.js/11.7.0/highlight.min.js"></script>
		<link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/highlight.js/11.7.0/styles/monokai.min.css">
		<style>
			body {
				font-family: sans-serif;
				margin: 1em;
				background-color: #2e2e2e;
				color: #eaeaea;
			}
			#prompt { width: 100%;
				box-sizing: border-box;
				background-color: #3d3d3d;
				color: #eaeaea;
				border: 1px solid #555;
				padding: 10px;
				border-radius: 5px;
			}
			#response {
				border: 1px solid #ccc;
				width: 100%;
				margin-top: 1em;
				padding: 0.5em;
				min-height: 100px;
				background-color: #3c3c3c;
				white-space: pre-wrap;
				border-radius: 5px;
			}
			.code-block {
				background-color: #383838;
				color: #f8f8f2;
				padding: 10px;
				border-radius:5px;
				overflow-x: auto;
			}
			pre code {
				background-color: #2c2c2c;
				color: #f8f8f2;
				padding: 10px;
				border-radius: 5px;
				overflow-x: auto;
			}
		</style>
	</head>
	<body>
		<h2>DeepSeek Chat VS Code Extension</h2>
		<textarea id="prompt" rows="3" placeholder="Ask something..."></textarea><br>
		<button id="askBtn" >Ask</button>
		<div id="response"></div>

		<script>
			const vscode = acquireVsCodeApi();

			document.getElementById('askBtn').addEventListener('click', () => {
				const text = document.getElementById('prompt').value;
				vscode.postMessage({ command: 'chat', text });
			});

			window.addEventListener('message', event => {
				const { command, text} = event.data;
				if (command === 'chatResponse') {
					const responseHtml = marked.parse(text);
					document.getElementById('response').innerHTML = responseHtml;
					hljs.highlightAll();
				}
			});
		</script>
	</body>
	</html>
	`
}

// This method is called when your extension is deactivated
export function deactivate() {}